package util

import (
	"aiRole/internal/dao/model"
	"github.com/gin-gonic/gin"
)

func GetUserInfo(c *gin.Context) *model.User {
	// 从上下文取出 userinfo
	val, exists := c.Get("userinfo")
	if !exists {
		// 没有用户信息，可以返回 nil
		return nil
	}

	// 尝试断言为 *model.User 指针
	if user, ok := val.(*model.User); ok {
		return user
	}

	// 如果存的是值类型 model.User，也可以处理
	if userVal, ok := val.(model.User); ok {
		return &userVal
	}

	// 类型不匹配，返回 nil
	return nil
}
